n = 3
num = [[0] * (n + 1) for _ in range(n + 1)]
[[0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0], [0, 0, 0, 0]]
print(num)
t = [0] * (n + 1)
print(t)
[0, 0, 0, 0]

def square_nums(nums):
    for i in nums:
        yield i**2


data = square_nums([1,2,3,4])
print(data)
print(next(data))
print("=========")
for i in data:
    print(i)


add = lambda a,b:a+b
print(add(3,4))

my_list = [1,2,3,4]
new_list = list(map(lambda x:x**2,my_list))
print(new_list)